#include "stdafx.h"
#include "pinyin.h"

pinyin::pinyin()
{

}

pinyin::~pinyin()
{

}

const char* pinyin::pinyin_[405] = 
{
	"a", "ai", "an", "ang", "ao", "ba", "bai", "ban", "bang", "bao", "bei", "ben", "beng", 
	"bi", "bian", "biao", "bie", "bin", "bing", "bo", "bu", "ca", "cai", "can", "cang", 
	"cao", "ce", "ceng", "cha", "chai", "chan", "chang", "chao", "che", "chen", "cheng", 
	"chi", "chong", "chou", "chu", "chuai", "chuan", "chuang", "chui", "chun", "chuo", 
	"ci", "cong", "cou", "cu", "cuan", "cui", "cun", "cuo", "da", "dai", "dan", "dang", 
	"dao", "de", "deng", "di", "dian", "diao", "die", "ding", "diu", "dong", "dou", "du", 
	"duan", "dui", "dun", "duo", "e", "en", "er", "fa", "fan", "fang", "fei", "fen", "feng", 
	"fo", "fou", "fu", "ga", "gai", "gan", "gang", "gao", "ge", "gei", "gen", "geng", "gong", 
	"gou", "gu", "gua", "guai", "guan", "guang", "gui", "gun", "guo", "ha", 
	"hai", "han", "hang", "hao", "he", "hei", "hen", "heng", "hong", "hou", "hu", "hua", "huai", 
	"huan", "huang", "hui", "hun", "huo", "ji", "jia", "jian", "jiang", "jiao", "jie", "jin", "jing", 
	"jiong", "jiu", "ju", "juan", "jue", "jun", "ka", "kai", "kan", "kang", "kao", "ke", "ken", "keng", 
	"kong", "kou", "ku", "kua", "kuai", "kuan", "kuang", "kui", "kun", "kuo", "la", "lai", "lan", "lang", 
	"lao", "le", "lei", "leng", "li", "lia", "lian", "liang", "liao", "lie", "lin", "ling", "liu", "long", 
	"lou", "lu", "lv", "luan", "lue", "lun", "luo", "ma", "mai", "man", "mang", "mao", "me", "mei", "men", 
	"meng", "mi", "mian", "miao", "mie", "min", "ming", "miu", "mo", "mou", "mu", "na", "nai", "nan", 
	"nang", "nao", "ne", "nei", "nen", "neng", "ni", "nian", "niang", "niao", "nie", "nin", "ning", 
	"niu", "nong", "nu", "nv", "nuan", "nue", "nuo", "o", "ou", "pa", "pai", "pan", "pang", "pao", 
	"pei", "pen", "peng", "pi", "pian", "piao", "pie", "pin", "ping", "po", "pu", "qi", "qia", "qian", 
	"qiang", "qiao", "qie", "qin", "qing", "qiong", "qiu", "qu", "quan", "que", "qun", "ran", "rang", 
	"rao", "re", "ren", "reng", "ri", "rong", "rou", "ru", "ruan", "rui", "run", "ruo", "sa", "sai", 
	"san", "sang", "sao", "se", "sen", "seng", "sha", "shai", "shan", "shang", "shao", "she", "shen", 
	"sheng", "shi", "shou", "shu", "shua", "shuai", "shuan", "shuang", "shui", "shun", "shuo", "si", 
	"song", "sou", "su", "suan", "sui", "sun", "suo", "ta", "tai", "tan", "tang", "tao", "te", "teng", 
	"ti", "tian", "tiao", "tie", "ting", "tong", "tou", "tu", "tuan", "tui", "tun", "tuo", "wa", "wai", 
	"wan", "wang", "wei", "wen", "weng", "wo", "wu", "xi", "xia", "xian", "xiang", "xiao", "xie", "xin", 
	"xing", "xiong", "xiu", "xu", "xuan", "xue", "xun", "ya", "yan", "yang", "yao", "ye", "yi", "yin", 
	"ying", "yo", "yong", "you", "yu", "yuan", "yue", "yun", "za", "zai", "zan", "zang", "zao", "ze", 
	"zei", "zen", "zeng", "zha", "zhai", "zhan", "zhang", "zhao", "zhe", "zhen", "zheng", "zhi", "zhong", 
	"zhou", "zhu", "zhua", "zhuai", "zhuan", "zhuang", "zhui", "zhun", "zhuo", "zi", "zong", "zou", "zu", 
	"zuan", "zui", "zun", "zuo"
};

int pinyin::pv_[396] = 
{
	-20319, -20317, -20304, -20295, -20292, -20283, -20265, -20257, -20242, -20230, -20051, 
	-20036, -20032, -20026, -20002, -19990, -19986, -19982, -19976, -19805, -19784, -19775, 
	-19774, -19763, -19756, -19751, -19746, -19741, -19739, -19728, -19725, -19715, -19540, 
	-19531, -19525, -19515, -19500, -19484, -19479, -19467, -19289, -19288, -19281, -19275, 
	-19270, -19263, -19261, -19249, -19243, -19242, -19238, -19235, -19227, -19224, -19218, 
	-19212, -19038, -19023, -19018, -19006, -19003, -18996, -18977, -18961, -18952, -18783, 
	-18774, -18773, -18763, -18756, -18741, -18735, -18731, -18722, -18710, -18697, -18696, 
	-18526, -18518, -18501, -18490, -18478, -18463, -18448, -18447, -18446, -18239, -18237, 
	-18231, -18220, -18211, -18201, -18184, -18183, -18181, -18012, -17997, -17988, -17970, 
	-17964, -17961, -17950, -17947, -17931, -17928, -17922, -17759, -17752, -17733, -17730, 
	-17721, -17703, -17701, -17697, -17692, -17683, -17676, -17496, -17487, -17482, -17468, 
	-17454, -17433, -17427, -17417, -17202, -17185, -16983, -16970, -16942, -16915, -16733, 
	-16708, -16706, -16689, -16664, -16657, -16647, -16474, -16470, -16465, -16459, -16452, 
	-16448, -16433, -16429, -16427, -16423, -16419, -16412, -16407, -16403, -16401, -16393, 
	-16220, -16216, -16212, -16205, -16202, -16187, -16180, -16171, -16169, -16158, -16155, 
	-15959, -15958, -15944, -15933, -15920, -15915, -15903, -15889, -15878, -15707, -15701, 
	-15681, -15667, -15661, -15659, -15652, -15640, -15631, -15625, -15454, -15448, -15436, 
	-15435, -15419, -15416, -15408, -15394, -15385, -15377, -15375, -15369, -15363, -15362, 
	-15183, -15180, -15165, -15158, -15153, -15150, -15149, -15144, -15143, -15141, -15140, 
	-15139, -15128, -15121, -15119, -15117, -15110, -15109, -14941, -14937, -14933, -14930, 
	-14929, -14928, -14926, -14922, -14921, -14914, -14908, -14902, -14894, -14889, -14882, 
	-14873, -14871, -14857, -14678, -14674, -14670, -14668, -14663, -14654, -14645, -14630, 
	-14594, -14429, -14407, -14399, -14384, -14379, -14368, -14355, -14353, -14345, -14170, 
	-14159, -14151, -14149, -14145, -14140, -14137, -14135, -14125, -14123, -14122, -14112, 
	-14109, -14099, -14097, -14094, -14092, -14090, -14087, -14083, -13917, -13914, -13910, 
	-13907, -13906, -13905, -13896, -13894, -13878, -13870, -13859, -13847, -13831, -13658, 
	-13611, -13601, -13406, -13404, -13400, -13398, -13395, -13391, -13387, -13383, -13367, 
	-13359, -13356, -13343, -13340, -13329, -13326, -13318, -13147, -13138, -13120, -13107, 
	-13096, -13095, -13091, -13076, -13068, -13063, -13060, -12888, -12875, -12871, -12860, 
	-12858, -12852, -12849, -12838, -12831, -12829, -12812, -12802, -12607, -12597, -12594, 
	-12585, -12556, -12359, -12346, -12320, -12300, -12120, -12099, -12089, -12074, -12067, 
	-12058, -12039, -11867, -11861, -11847, -11831, -11798, -11781, -11604, -11589, -11536, 
	-11358, -11340, -11339, -11324, -11303, -11097, -11077, -11067, -11055, -11052, -11045, 
	-11041, -11038, -11024, -11020, -11019, -11018, -11014, -10838, -10832, -10815, -10800, 
	-10790, -10780, -10764, -10587, -10544, -10533, -10519, -10331, -10329, -10328, -10322, 
	-10315, -10309, -10307, -10296, -10281, -10274, -10270, -10262, -10260, -10256, -10254
};

void
pinyin::first_letter(int code, std::string& strletter)
{
	if(code >= 1601 && code < 1637) strletter = _T("A");
	if(code >= 1637 && code < 1833) strletter = _T("B");
	if(code >= 1833 && code < 2078) strletter = _T("C");
	if(code >= 2078 && code < 2274) strletter = _T("D");
	if(code >= 2274 && code < 2302) strletter = _T("E");
	if(code >= 2302 && code < 2433) strletter = _T("F");
	if(code >= 2433 && code < 2594) strletter = _T("G");
	if(code >= 2594 && code < 2787) strletter = _T("H");
	if(code >= 2787 && code < 3106) strletter = _T("J");
	if(code >= 3106 && code < 3212) strletter = _T("K");
	if(code >= 3212 && code < 3472) strletter = _T("L");
	if(code >= 3472 && code < 3635) strletter = _T("M");
	if(code >= 3635 && code < 3722) strletter = _T("N");
	if(code >= 3722 && code < 3730) strletter = _T("O");
	if(code >= 3730 && code < 3858) strletter = _T("P");
	if(code >= 3858 && code < 4027) strletter = _T("Q");
	if(code >= 4027 && code < 4086) strletter = _T("R");
	if(code >= 4086 && code < 4390) strletter = _T("S");
	if(code >= 4390 && code < 4558) strletter = _T("T");
	if(code >= 4558 && code < 4684) strletter = _T("W");
	if(code >= 4684 && code < 4925) strletter = _T("X");
	if(code >= 4925 && code < 5249) strletter = _T("Y");
	if(code >= 5249 && code < 5590) strletter = _T("Z");
}

void
pinyin::get_first_letter(std::string strname, std::string& strfirstletter)
{
	BYTE ucHigh, ucLow;
	int  nCode;
	std::string strRet;
	strfirstletter = "";

	for (int i=0; i<strname.size(); i++)
	{
		if ( (BYTE)strname[i] < 0x80 )
		{
			strfirstletter += strname[i];
			continue;
		}

		ucHigh = (BYTE)strname[i];
		ucLow  = (BYTE)strname[i+1];
		if ( ucHigh < 0xa1 || ucLow < 0xa1)
		{
			
			continue;
		}
		else
		{
			// Treat code by section-position as an int type parameter,
			// so make following change to nCode.
			nCode = (ucHigh - 0xa0) * 100 + ucLow - 0xa0;
			first_letter(nCode, strRet);
			strfirstletter += strRet;
			i++;
		}
	}
}

bool
pinyin::find_str(std::string str, std::string find)
{
	std::string srcpy;
	get_first_letter(str, srcpy);

	std::string strToFindPY;
	strToFindPY = find;
	aot::tt::utility::StrUtil::ToUpperCase(strToFindPY);
	//strToFindPY = StrToUpper(strToFindPY);
	//GetFirstLetter(strToFind, strToFindPY);
	return (srcpy.find(strToFindPY) != -1);
}

inline int 
pinyin::china_code_value(const std::string& word)   
{
	char code[11];   
	memset(code, 0, sizeof(code));   
	code[0] = word[1];
	code[1] = word[0];
	return (*(long*)code) - 65536;   
}

inline std::string 
pinyin::code_to_py(int china_code)   
{   
	std::string py_mode;
	if(china_code < -20319 || china_code > -10247)
	{
		py_mode += "";
	}
	else
	{  
		int   pos;
		for(pos = sizeof(pv_)/sizeof(int) - 1; pos >= 0; --pos)
			if   (pv_[pos] <= china_code)   break;   
		py_mode += pinyin_[pos];   
		py_mode += "";   
	}   

	return py_mode;
}   

std::string 
pinyin::china_to_py(const std::string& str)
{
	BYTE ucHigh, ucLow;
	char   code[11];
	std::string strpy,strRet;
	strRet = "";

	for (int i=0; i<str.size(); i++)
	{
		if ( (BYTE)str[i] < 0x80 )
		{
			strRet += str[i];
			continue;
		}

		ucHigh = (BYTE)str[i];
		ucLow  = (BYTE)str[i+1];
		if ( ucHigh < 0xa1 || ucLow < 0xa1 )
		{
			continue;
		}
		else
		{
			memset(code, 0, sizeof(code)); 
			code[0] = str[i+1];
			code[1] = str[i];
			int ncode = (*(long*)code) - 65536;
			strpy = code_to_py(ncode);
			strRet += strpy;
			i++;
		}
	}
	return strRet;
}